#pragma once

#include <signal.h>
#include <functional>
#include "Socket.hpp"

class HttpServer
{
// public:
    using func_t = std::function<void(int)>;
public:
    HttpServer(uint16_t port, func_t func, const std::string& ip = "")
        :_func(func)
    {
        _listensock = Sock::Socket();
        Sock::Bind(_listensock, port, ip);
        Sock::Listen(_listensock);
    }

    void start()
    {        
        signal(SIGCHLD, SIG_IGN);
        for(;;)
        {
            int serverfd = Sock::Accept(_listensock, nullptr, nullptr);
            if(serverfd < 0) continue;
            // 多进程版本
            if(fork() == 0)
            {
                // child
                close(_listensock);
                _func(serverfd);
                close(serverfd);
                exit(0);
            }
            close(serverfd);
            // sleep(1);
        }

    }

    ~HttpServer()
    {
        if(_listensock >= 0)
        {
            close(_listensock);
        }
    }

private:
    int _listensock;
    func_t _func;
};